package com.mo.axion.web.service.impl;

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.text.CharSequenceUtil;
import cn.hutool.core.util.ObjectUtil;
import com.mo.axion.common.constant.AxionConstant;
import com.mo.axion.common.constant.CommonConstant;
import com.mo.axion.common.enums.AuditStatus;
import com.mo.axion.common.enums.MessageStatus;
import com.mo.axion.common.enums.RespStatusEnum;
import com.mo.axion.common.enums.TemplateType;
import com.mo.axion.common.vo.BasicResultVO;
import com.mo.axion.cron.xxl.entity.XxlJobInfo;
import com.mo.axion.cron.xxl.service.CronTaskService;
import com.mo.axion.cron.xxl.utils.XxlJobUtils;
import com.mo.axion.support.dao.MessageTemplateDao;
import com.mo.axion.support.domain.MessageTemplate;
import com.mo.axion.web.service.MessageTemplateService;
import com.mo.axion.web.vo.MessageTemplateParam;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.stereotype.Service;

import javax.persistence.criteria.Predicate;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.Optional;

/**
 * @author Mo
 * @createTime 2024/4/8 17:18
 * @description 消息模板管理接口实现类
 */
@Service
public class MessageTemplateServiceImpl implements MessageTemplateService {

    @Autowired
    private MessageTemplateDao messageTemplateDao;

    @Autowired
    private CronTaskService cronTaskService;

    @Autowired
    private XxlJobUtils xxlJobUtils;

    /**
     * 分页查询未删除的模板列表
     *
     * @param messageTemplateParam
     * @return
     */
    @Override
    public Page<MessageTemplate> queryList(MessageTemplateParam messageTemplateParam) {
        PageRequest pageRequest = PageRequest.of(messageTemplateParam.getPage() - 1, messageTemplateParam.getPerPage());
        String creator = CharSequenceUtil.isBlank(messageTemplateParam.getCreator()) ? AxionConstant.DEFAULT_CREATOR : messageTemplateParam.getCreator();
        return messageTemplateDao.findAll((Specification<MessageTemplate>) (root, query, cb) -> {
            List<Predicate> predicateList = new ArrayList<>();
            // 加搜索条件
            if (CharSequenceUtil.isNotBlank(messageTemplateParam.getKeywords())) {
                predicateList.add(cb.like(root.get("name").as(String.class), "%" + messageTemplateParam.getKeywords() + "%"));
            }
            predicateList.add(cb.equal(root.get("isDeleted").as(Integer.class), CommonConstant.FALSE));
            predicateList.add(cb.equal(root.get("creator").as(String.class), creator));
            Predicate[] p = new Predicate[predicateList.size()];
            // 查询
            query.where(cb.and(predicateList.toArray(p)));
            // 排序
            query.orderBy(cb.desc(root.get("updated")));
            return query.getRestriction();
        }, pageRequest);
    }

    /**
     * 统计未删除的条数
     *
     * @return
     */
    @Override
    public Long count() {
        return messageTemplateDao.countByIsDeletedEquals(CommonConstant.FALSE);
    }

    @Override
    public MessageTemplate saveOrUpdate(MessageTemplate messageTemplate) {
        //  有模板Id则更新，无模板Id则插入
        if (Objects.isNull(messageTemplate.getId())) {
            initStatus(messageTemplate);
        } else {
            resetStatus(messageTemplate);
        }
        messageTemplate.setUpdated(Math.toIntExact(DateUtil.currentSeconds()));
        return messageTemplateDao.save(messageTemplate);
    }

    /**
     * 软删除(deleted=1)
     *
     * @param ids 要删除的模板IdList
     */
    @Override
    public void deleteByIds(List<Long> ids) {
        Iterable<MessageTemplate> messageTemplates = messageTemplateDao.findAllById(ids);
        messageTemplates.forEach(messageTemplate -> messageTemplate.setIsDeleted(CommonConstant.TRUE));
        for (MessageTemplate messageTemplate : messageTemplates) {
            if (Objects.nonNull(messageTemplate.getCronTaskId()) && messageTemplate.getCronTaskId() > 0) {
                cronTaskService.deleteCronTask(messageTemplate.getCronTaskId());
            }
        }
        messageTemplateDao.saveAll(messageTemplates);
    }

    /**
     * 根据ID查询模板信息
     *
     * @param id 模板Id
     * @return
     */
    @Override
    public MessageTemplate queryById(Long id) {
        return messageTemplateDao.findById(id).orElse(null);
    }

    /**
     * 复制配置
     *
     * @param id
     */
    @Override
    public void copy(Long id) {
        MessageTemplate messageTemplate = messageTemplateDao.findById(id).orElse(null);
        if (Objects.nonNull(messageTemplate)) {
            MessageTemplate clone = ObjectUtil.clone(messageTemplate).setId(null).setCronTaskId(null);
            messageTemplateDao.save(clone);
        }
    }

    /**
     * 启动模板的定时任务
     *
     * @param id 模板Id
     * @return 响应
     */
    @Override
    public BasicResultVO startCronTask(Long id) {
        //  1. 获取消息模板信息
        MessageTemplate messageTemplate = messageTemplateDao.findById(id).orElse(null);
        if (Objects.isNull(messageTemplate)) {
            return BasicResultVO.fail();
        }
        //  2. 动态创建或更新定时任务
        XxlJobInfo xxlJobInfo = xxlJobUtils.buildXxlJobInfo(messageTemplate);
        //  3. 获取taskId(如果本身存在则复用原有任务，如果不存在则得到新建后任务Id)
        Integer taskId = messageTemplate.getCronTaskId();
        BasicResultVO basicResultVO = cronTaskService.saveCronTask(xxlJobInfo);
        if (
                Objects.isNull(taskId)
                && RespStatusEnum.SUCCESS.getCode().equals(basicResultVO.getStatus())
                && Objects.nonNull(basicResultVO.getData())
        ) {
             taskId = Integer.valueOf(String.valueOf(basicResultVO.getData()));
        }
        //  4. 启动定时任务
        if (Objects.nonNull(taskId)) {
            cronTaskService.startCronTask(taskId);
            MessageTemplate clone = ObjectUtil.clone(messageTemplate).setMsgStatus(MessageStatus.RUN.getCode()).setCronTaskId(taskId).setUpdated(Math.toIntExact(DateUtil.currentSeconds()));
            messageTemplateDao.save(clone);
            return BasicResultVO.success();
        }
        return BasicResultVO.fail();
    }

    /**
     * 暂停模板的定时任务
     *
     * @param id 模板Id
     * @return 响应
     */
    @Override
    public BasicResultVO stopCronTask(Long id) {
        //  1. 修改模板状态
        MessageTemplate messageTemplate = messageTemplateDao.findById(id).orElse(null);
        if (Objects.isNull(messageTemplate)) {
            return BasicResultVO.fail();
        }

        MessageTemplate clone = ObjectUtil.clone(messageTemplate)
                .setMsgStatus(MessageStatus.STOP.getCode())
                .setUpdated(Math.toIntExact(DateUtil.currentSeconds()));
        messageTemplateDao.save(clone);
        //  暂停定时任务
        return cronTaskService.stopCronTask(clone.getCronTaskId());
    }

    /**
     * 初始化模板信息
     * @param messageTemplate
     */
    private void initStatus(MessageTemplate messageTemplate) {
        messageTemplate.setFlowId(CharSequenceUtil.EMPTY)
                .setMsgStatus(MessageStatus.INIT.getCode())
                .setAuditStatus(AuditStatus.WAIT_AUDIT.getCode())
                .setCreator(CharSequenceUtil.isBlank(messageTemplate.getCreator()) ?
                        AxionConstant.DEFAULT_CREATOR : messageTemplate.getCreator())
                .setUpdator(CharSequenceUtil.isBlank(messageTemplate.getUpdator()) ?
                        AxionConstant.DEFAULT_UPDATOR : messageTemplate.getUpdator())
                .setTeam(CharSequenceUtil.isBlank(messageTemplate.getTeam()) ?
                        AxionConstant.DEFAULT_TEAM : messageTemplate.getTeam())
                .setAuditor(CharSequenceUtil.isBlank(messageTemplate.getAuditor()) ?
                        AxionConstant.DEFAULT_AUDITOR : messageTemplate.getAuditor())
                .setCreated(Math.toIntExact(DateUtil.currentSeconds()))
                .setIsDeleted(CommonConstant.FALSE);
    }

    /**
     * 重置模板状态
     * 修改定时任务信息
     * @param messageTemplate
     */
    private void resetStatus(MessageTemplate messageTemplate) {
        messageTemplate.setUpdator(messageTemplate.getUpdator())
                .setMsgStatus(MessageStatus.INIT.getCode())
                .setAuditStatus(AuditStatus.WAIT_AUDIT.getCode());

        //  从数据库查询并注入 定时任务Id
        MessageTemplate dbMsg = queryById(messageTemplate.getId());
        if (Objects.nonNull(dbMsg) && Objects.nonNull(dbMsg.getCronTaskId())) {
            messageTemplate.setCronTaskId(dbMsg.getCronTaskId());
        }
        if (Objects.nonNull(messageTemplate.getCronTaskId()) && TemplateType.CLOCKING.getCode().equals(messageTemplate.getTemplateType())) {
            XxlJobInfo xxlJobInfo = xxlJobUtils.buildXxlJobInfo(messageTemplate);
            cronTaskService.saveCronTask(xxlJobInfo);
            cronTaskService.stopCronTask(messageTemplate.getCronTaskId());
        }
    }
}
